import argparse

def read_args():
	parser = argparse.ArgumentParser()
	parser.add_argument('--train_split_size', type = float, default = 0.8,
				   help='split rate')
	parser.add_argument('--if_constrained', type = str, default = 'unconstrained',
				   help='constrained or unconstrained')
	parser.add_argument('--input_size', type=int, default=27, #原来27
						help='model input size')
	parser.add_argument('--num_lstm_layers', type=int, default=2, # 原来是4
						help='model num_lstm_layers')
	parser.add_argument('--lstm_size', type=int, default=15, # 原来1024
						help='model lstm_size')
	parser.add_argument('--dropout', type=float, default=0.2,
						help='model dropout')
	parser.add_argument('--loss', type=str, default='else',
						help='penalty or else')
	parser.add_argument('--penalty', type=float, default=1.0,
						help='penalty num')
	parser.add_argument('--device', type=str, default='cuda',
						help='cuda or cup')
	parser.add_argument('--batch_size', type=int, default=64,
						help='batch_size')
	parser.add_argument('--num_epoch', type=int, default=2000, # 原来是1000效果比较好, 2000效果最好
						help='num_epoch')
	parser.add_argument('--learning_rate', type=float, default=0.05, #原来是0.01，0.0001比较好
						help='learning_rate')
	parser.add_argument('--scheduler_step_size', type=int, default=500,
						help='scheduler_step_size')




	args = parser.parse_args()

	return args
